Crate burn_jit

Burn JIT Backend

Modules

• compute
  Compute related module.
• gpu
  GPU dialect module that contains a representation that can be used to program any GPU.
• kernel
  Kernel module
• tensor
  Tensor module.

Macros

• binary
  Creates a binary kernel.
• unary
  Creates a unary kernel.

Structs

• JitBackend
  Generic tensor backend that can be compiled just-in-time to any shader runtime
• PrecisionBridge
  Handle precision conversion for the jit backend.

Traits

• Compiler
  Compiles the gpu representation into its own representation that can be formatted into tokens.
• FloatElement
  The float element type for the jit backend.
• IntElement
  The int element type for the jit backend.
• JitElement
  The base element trait for the jit backend.
• Runtime
  Runtime for the just-in-time backend.

Type Aliases

• RuntimeInt
  Type alias to the runtime signed int element type.